package cn.com.jrr.core.freemarker;

import java.io.BufferedWriter;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.Map;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

public class FreemarkerUtil {
    public Template getTemplate(String name) {
        try {
            // 通过Freemaker的Configuration读取相应的ftl
            Configuration cfg = new Configuration();
            // 设定去哪里读取相应的ftl模板文件
            cfg.setClassForTemplateLoading(this.getClass(), "/ftl");
            cfg.setDefaultEncoding("UTF-8");
            // 在模板文件目录中找到名称为name的文件
            Template temp = cfg.getTemplate(name,"UTF-8");
            return temp;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 控制台输出
     *
     * @param name
     * @param root
     */
    public void print(String name, Map<String, Object> root) {
        try {
            // 通过Template可以将模板文件输出到相应的流
            Template temp = this.getTemplate(name);
            temp.process(root, new PrintWriter(System.out));
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 输出HTML文件
     *
     * @param name
     * @param root
     * @param outFile
     */
    public void fprint(String name, Map<String, Object> root, String outFile) {
        FileOutputStream fos;
        OutputStreamWriter osw = null ;
        BufferedWriter bw = null ;
        try {
            // 通过一个文件输出流，就可以写到相应的文件中，此处用的是绝对路径
            fos= new FileOutputStream(outFile);
            osw =new OutputStreamWriter(fos, "UTF-8");
            bw =new BufferedWriter(osw, 1024);
            Template temp = this.getTemplate(name);
            temp.process(root, bw);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TemplateException e) {
            e.printStackTrace();
        } finally {
            try {
                if (bw != null){
                    bw.close();
                }
                if(osw !=null){
                    osw.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}
